import os
from markitdown import MarkItDown
from openai import OpenAI

# --- 您的原始配置 ---
client = OpenAI(api_key="sk-OSErNUxQt3yVhPHv6a309a382eCa4e5aB978F921D1C5B808", base_url="http://106.54.60.26:3400/v1")
md = MarkItDown(llm_client=client, llm_model="gemini-2.5-flash-preview-05-20")

# --- 处理并保存输出 ---

# 1. 定义输出文件名和图片存放目录
output_md_file = "output.md"
image_save_directory = "output_images" # 图片将自动保存到这里

# 2. 确保图片目录存在
os.makedirs(image_save_directory, exist_ok=True)

# 3. 执行转换，并使用 image_dir 参数指定图片保存位置
print("正在转换文档，图片将自动保存...")
result = md.convert(
    r"C:\Users\Administrator\Documents\（任务资料）商品信息.docx",
    image_dir=image_save_directory  # <-- 关键参数在这里！
)
print("文档转换完成！")


# 4. 直接保存Markdown文本内容
# 库会自动在文本中生成指向 'output_images' 文件夹中对应图片的正确链接
with open(output_md_file, "w", encoding="utf-8") as f:
    f.write(result.text_content)

print(f"\n转换成功！")
print(f"Markdown文件已保存为: {output_md_file}")
print(f"所有图片已自动保存到: {image_save_directory}")